#!/usr/bin/env python
# -*- encoding: utf-8 -*-
'''
@File    :   visualize.py
@Time    :   2021/11/06 20:21:46
@Author  :   Yaadon 
'''

# here put the import lib
import matplotlib.pyplot as plt

def visualize(srcimg, img_enhance):
    """
    @Description : 定义可视化函数，用于对比原图和图像增强的效果
    ------------
    @Args : srcimg:原图 img_enhance:修改后图
    -----
    @Returns : 
    --------
    """
    # 图像可视化
    plt.figure(num=2, figsize=(6,12))
    plt.subplot(1,2,1)
    plt.title('Src Image', color='#0000FF')
    plt.axis('off') # 不显示坐标轴
    plt.imshow(srcimg) # 显示原图片

    plt.subplot(1,2,2)
    plt.title('Enhance Image', color='#0000FF')
    plt.axis('off') # 不显示坐标轴
    plt.imshow(img_enhance)
    plt.show()
